package com.sweet.controller;

import com.alibaba.csp.sentinel.annotation.SentinelResource;
import com.alibaba.csp.sentinel.slots.block.BlockException;
import com.sweet.Result;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.concurrent.TimeUnit;

@RestController
@Slf4j
public class FlowLimitController {
    @GetMapping("/testA")
    public String testA() {
        return "------testA";
    }

    @GetMapping("/testB")
    public String testB() {
        log.info(Thread.currentThread().getName() + "-----testB");
        return "------testB";
    }

    @GetMapping("/testC")
    public String testC() {
        log.info("test ------ RT");
        try {
            TimeUnit.SECONDS.sleep(1);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return "------testC";
    }

    @GetMapping("/testD")
    public String testD() {
        log.info("test ------ 异常比例");
        int a = 10/0;
        return "------testD";
    }

    @GetMapping("/testE")
    public String testE() {
        log.info("test ------ 异常数");
        int a = 10/0;
        return "------testE";
    }

    @GetMapping("/testHotKey")
    @SentinelResource(value="testHotKey", blockHandler = "deal_testHotKey")
    public Result testHotKey(@RequestParam(value = "p1", required = false) String p1,
                             @RequestParam(value = "p2",required = false) String p2) {
        log.info("test ------ testHotKey");
        return Result.ok("testHotKey请求成功!");
    }

    public Result deal_testHotKey(String p1, String p2, BlockException e) {
        log.error("deal ------ testHotKey");
        log.error(e.getMessage());
        return Result.error("请求超时亲稍后再试~~");
    }
}
